#include <iostream>
#include <stdarg.h>
#include <string>
#include <ctime>
#include <unistd.h>
#include <fstream>
#include <sys/types.h>
#include <sys/stat.h>
#include <fcntl.h>
using namespace std;

// 定义一个枚举，表示日志级别
enum
{
    Debug = 0,
    Info,
    Warning,
    Error,
    Fatal
};

// 定义一个枚举，表示输出方式
enum
{
    Screen = 10,
    OneFile,
    ClassFile
};

// 定义一个函数，将日志级别转换为字符串
string Leveltostring(int level)
{
    switch (level)
    {
    case Debug:
        return "Debug";
    case Info:
        return "Info";
    case Warning:
        return "Warning";
    case Error:
        return "Error";
    case Fatal:
        return "Fatal";
    default:
        return "Unknown";
    }
}

const int default_style = Screen;       // 默认想显示器打印
const string default_filename = "log."; // 默认的文件名是log.
const string logdir="log";
// 定义一个日志类
class Log
{
public:
    Log() : style(default_style), filename(default_filename)
    {
        mkdir(logdir.c_str(), 0777);// 创建一个目录
    }
    // 定义一个函数，将时间戳转换为字符串
    string Timelocaltime()
    {
        time_t curtime = time(nullptr);
        struct tm *curr = localtime(&curtime);

        char time_buffer[128];
        snprintf(time_buffer, sizeof(time_buffer), "%d年-%d月-%d日 %d时:%d分:%d秒",
                 curr->tm_year + 1900, curr->tm_mon + 1, curr->tm_mday, curr->tm_hour, curr->tm_min, curr->tm_sec);
        return time_buffer;
    }

    // 定义一个函数，设置一个style，向哪里输出，默认是向屏幕输出
    void SetStyle(int style) // 设置一个style，向哪里输出，默认是向屏幕输出
    {
        this->style = style;
    }
    // 定义一个函数，设置一个文件名
    void SetFilename(const string &filename)
    {
        this->filename = filename;
    }

    // 定义一个函数，将日志写入文件
    void WriteLogToOneFile(const string &logname, const string &message)
    {
        int fd = open(logname.c_str(), O_CREAT | O_WRONLY | O_APPEND, 0666);
        if (fd < 0)
        {
            exit(-1);
        }

        write(fd, message.c_str(), message.size());
        close(fd);
    }

    // 定义一个函数，将日志写入文件
    void WriteLogToClassFile(const string &levelstr, const string &message) // 将日志写入文件
    {
        string logname = logdir;
        logname += "/";
        logname += filename;
        logname += levelstr;
        WriteLogToOneFile(logname, message);
    }

    //
    void WriteLog(const string &levelstr, const string &message)
    {

        switch (style)
        {
        case Screen:
            cout << message;
            break;
        case OneFile:
            WriteLogToClassFile("all", message);
            break;
        case ClassFile:
            WriteLogToClassFile(levelstr, message);
            break;
        default:
            break;
        }
    }
    void LogMessage(int level, const char *format, ...)
    {
        char rightbuffer[1024]; // 日志的内容
        va_list args;           // va_list其实就是char* 类型的
        va_start(args, format); // 获取可变参数的位置,由args去指向可变参数部分
        vsnprintf(rightbuffer, sizeof(rightbuffer), format, args);
        va_end(args); // 相等于args=nullptr

        char leftbuffer[1024];
        string levelstr = Leveltostring(level);
        string curtime = Timelocaltime();
        string idstr = to_string(getpid());
        snprintf(leftbuffer, sizeof(leftbuffer), "[%s][%s][%s]", levelstr.c_str(), curtime.c_str(), idstr.c_str());

        string loginfo = leftbuffer;
        loginfo.append(rightbuffer);

        WriteLog(levelstr, loginfo);
    }

    ~Log()
    {
    }

private:
    int style;
    string filename;
};